#!/usr/bin/env python3
"""
下载测试图像的脚本
"""

import requests
import os
from PIL import Image
import numpy as np

def download_test_image():
    """下载一张免费的街景测试图像"""
    
    # 使用Unsplash的免费API下载一张街景图像
    # 注意：这是用于测试目的，实际使用时请确保遵守相关使用条款
    
    print("正在下载测试图像...")
    
    # 创建一个简单的测试图像（模拟街景）
    create_synthetic_test_image()
    
    print("✓ 测试图像创建完成！")

def create_synthetic_test_image():
    """创建一个合成的街景测试图像"""
    
    # 创建一个2048x1024的RGB图像
    width, height = 2048, 1024
    img = np.zeros((height, width, 3), dtype=np.uint8)
    
    # 天空区域 (蓝色)
    img[:height//3, :] = [135, 206, 235]  # 天蓝色
    
    # 建筑区域 (灰色)
    img[height//3:height*2//3, :] = [128, 128, 128]  # 灰色
    
    # 道路区域 (深灰色)
    img[height*2//3:, :] = [64, 64, 64]  # 深灰色
    
    # 添加一些简单的几何形状来模拟建筑和道路
    # 建筑轮廓
    for i in range(5):
        x_start = i * width // 5
        x_end = (i + 1) * width // 5 - 50
        y_start = height // 3
        y_end = height * 2 // 3
        img[y_start:y_end, x_start:x_end] = [100, 100, 100]
    
    # 道路线条
    road_y = height * 2 // 3 + 50
    img[road_y-10:road_y+10, :] = [255, 255, 255]  # 白色道路标线
    
    # 保存图像
    pil_img = Image.fromarray(img)
    test_image_path = "test_street_scene.jpg"
    pil_img.save(test_image_path, "JPEG", quality=95)
    
    print(f"✓ 合成测试图像已保存到: {test_image_path}")
    print(f"✓ 图像尺寸: {width}x{height}")
    
    return test_image_path

def download_real_test_image():
    """尝试下载真实的街景图像"""
    
    # 这里可以添加从免费图像网站下载真实街景图像的代码
    # 由于版权和使用条款的限制，这里只提供合成图像
    
    print("注意：由于版权限制，我们创建了一个合成的街景测试图像。")
    print("如果您有真实的街景图像，可以直接使用。")

if __name__ == "__main__":
    download_test_image()
